module pc_gen#(
  parameter BOOT_ADDR = 32'b0
) (
  input clk,
  input rst_n,
  input      [31:0] pc_current,
  input             pc_stall,
  input             pc_jmp_ena,
  input      [31:0] pc_jmp_addr,
  output reg [31:0] pc_next
);

  wire [31:0] pc_inc;

  assign pc_inc = pc_current + 32'h4;

  always @(*) begin
    pc_next = pc_inc;
    if (pc_stall) begin
      pc_next = pc_current;
    end else if (pc_jmp_ena) begin
      pc_next = pc_jmp_addr;
    end
  end

endmodule

